how to write for loop in python